package com.briup.mt.web.controller;

import com.briup.mt.bean.Module;
import com.briup.mt.service.IGreenHouseMessageService;
import com.briup.mt.service.IGreenHouseModuleService;
import com.briup.mt.utils.ExcelUtils;
import com.briup.mt.utils.Message;
import com.briup.mt.utils.MessageUtil;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import javax.validation.constraints.NotNull;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author yxy
 * @version 1.0
 * @date 2020/12/16 22:28
 */
@RestController
@Validated
@RequestMapping("/greenhouse_module")
public class GreenHouseModuleController {
    @Resource
    private IGreenHouseModuleService iGreenHouseModuleService;
    @ApiOperation(value = "导入到Excel中", notes = "注意！测试的时候请将地址粘贴到浏览器地址栏测试", produces = "module/octet-stream")
    @GetMapping("download")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "name", value = "应用名称", paramType = "query")
    })
    public void download(HttpServletResponse response, String name) throws Exception {
        // 2. 将excel导出
        // 2.1 excel头部数据
        String[] headList = new String[] { "序号", "名称", "运算法则", "介绍","状态", "发布时间", "应用序列号", "用户Id","资料Id"};
        // 2.2 属性
        String[] fieldList = new String[] { "id", "name", "algorithm", "introduce","status", "create_time",
                "appliction_id", "user_id","dataset_id"};
        // 2.3 数据列，需要转化
        List<Map<String, Object>> dataList = new ArrayList<>();
        List<Module> list= iGreenHouseModuleService.query(name);
        for (Module module : list) {
            Map<String, Object> map = new HashMap<>();
            map.put("id", module.getId());
            map.put("name", module.getName());
            map.put("algorithm",module.getAlgorithm());
            map.put("introduce", module.getIntroduce());
            map.put("status", module.getStatus());
            map.put("create_time", module.getCreateTime());
            map.put("appliction_id", module.getApplicationId());
            map.put("user_id", module.getUserId());
            map.put("dataset_id", module.getDatasetId());
            dataList.add(map);
        }
        // 2.4 导出文件名
        String excelName = "module_list";
        // 3. 调用工具方法导出Excel
        ExcelUtils.createExcel(response, excelName, headList, fieldList, dataList);

    }
    @GetMapping("query")
    public Message query(String name){
        List<Module> list = iGreenHouseModuleService.query(name);
        return MessageUtil.success(list);
    }

    @PostMapping("saveOrUpdate")
    public Message saveOrUpdate(Module info){
        iGreenHouseModuleService.saveOrUpdate(info);
        return MessageUtil.success("操作成功");
    }

    @DeleteMapping("deleteByID")
    @ApiImplicitParams({@ApiImplicitParam(name = "id",value="用户编号",required =true, paramType = "query"),})
    public Message deleteByID(@NotNull long id){
        iGreenHouseModuleService.deleteById(id);
        return MessageUtil.success("操作成功");
    }
}
